import os

def mypathsize(mypath):
	'''
	定义统计指定目录大小的函数
	'''
	size=0
	
	mypathList = os.listdir(mypath) #列出目标目录的文件列表
	if mypathList != []:		#如果文件列表为空，就跳过文件大小检测
		for i in mypathList:		
			filepath = os.path.join(mypath,i)		#得到目标的实际地址
			if os.path.isdir(filepath):				#当目标为文件夹时
				size += mypathsize(filepath)		#递归统计，再次在filepath位置调用函数
			if os.path.isfile(filepath):			#当目标为文件时
				size += os.path.getsize(filepath)
	return size

while True:		#判断循环，用来输入信息
	print("{0:5}{1:5}".format('1、输入文件目录','2、退出'))
	key=input(">请输入对应的选择：")

	if key=="1":
		add_r=input("请输入目录或文件地址:")
		print("{}字节".format(mypathsize(add_r)))
		input("按回车键继续：")
	elif key=="2":
		print("已退出")
		break
	else:
		print("输入有误")
